package com.baidu.automata.factory;

import java.io.File;
import java.util.Iterator;

import javax.xml.parsers.SAXParser;
import javax.xml.parsers.SAXParserFactory;

import org.dom4j.Document;
import org.dom4j.Element;
import org.dom4j.io.SAXReader;
import org.dom4j.io.SAXValidator;
import org.dom4j.util.XMLErrorHandler;

import com.baidu.automata.core.context.Context;
import com.baidu.common.util.FieldUtil;
import com.baidu.config.CtxConfig;

public class CtxFactory {
	private String ctxId = null;

	private String ctxFilePath = null;

	private XMLErrorHandler errorHandler = new XMLErrorHandler();

	private SAXParser parser = null;

	private Document xmlDocument = null;

	private void setDocument() throws Exception {
		// 获取基于 SAX 的解析器的实例
		SAXParserFactory factory = SAXParserFactory.newInstance();
		// 解析器在解析时验证 XML 内容。
		factory.setValidating(true);
		// 指定由此代码生成的解析器将提供对 XML 名称空间的支持。
		factory.setNamespaceAware(true);
		// 使用当前配置的工厂参数创建 SAXParser 的一个新实例。
		parser = factory.newSAXParser();

		SAXReader xmlReader = new SAXReader();

		xmlDocument = (Document) xmlReader.read(new File(ctxFilePath));
	}

	private void setCtxFilePath() throws Exception {
		this.ctxFilePath = CtxConfig.getInstance().getProperty(this.ctxId);
	}

	private void validatorXml() throws Exception {
		// 设置 XMLReader 的基础实现中的特定属性。
		parser.setProperty("http://java.sun.com/xml/jaxp/properties/schemaLanguage", "http://www.w3.org/2001/XMLSchema");
		
		// 创建一个SAXValidator校验工具，并设置校验工具的属性
		SAXValidator validator = new SAXValidator(parser.getXMLReader());
		// 设置校验工具的错误处理器，当发生错误时，可以从处理器对象中得到错误信息。
		validator.setErrorHandler(errorHandler);
		// 校验
		validator.validate(xmlDocument);

		// 如果错误信息不为空，说明校验失败，打印错误信息
		if (errorHandler.getErrors().hasContent()) {
			throw new Exception(errorHandler.getErrors().asXML());
		}
	}

	@SuppressWarnings("unchecked")
	private Context initCtx() throws Exception {
		Element root = xmlDocument.getRootElement();
		String ctxId = root.attributeValue("id");
		String p = root.attributeValue("parent");

		Context c = new Context(ctxId);

		for (Iterator<Element> it = root.elementIterator(); it.hasNext();) {
			Element e = it.next();
			c.addDataField(FieldUtil.initFiled(e));
		}

		if (p != null && !"".equals(p)) {
			c.setParent(getCtxById(p));
		}
		
		return c;
	}

	public static Context getCtxById(String id) throws Exception {
		CtxFactory f = new CtxFactory();
		f.ctxId = id;
		f.setCtxFilePath();
		f.setDocument();
		f.validatorXml();

		return f.initCtx();
	}
}
